clu
git+https://github.com/deepmind/einshape
jax>=0.4.7
jaxlib>=0.4
flax>=0.6.8
ml_collections
numpy>=1.21.4
opencv-python>=4.4.0
optax
scikit-image==0.17.2
tensorboard>=2.4.0
